forked from x/icebergs
78 lines
2.7 KiB
Go
78 lines
2.7 KiB
Go
package crx
|
|
|
|
import (
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/mdb"
|
|
"github.com/shylinux/icebergs/base/nfs"
|
|
"github.com/shylinux/icebergs/base/web"
|
|
kit "github.com/shylinux/toolkits"
|
|
|
|
"github.com/nareix/joy4/av"
|
|
"github.com/nareix/joy4/av/avutil"
|
|
|
|
"path"
|
|
)
|
|
|
|
const CACHE = "cache"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
CACHE: {Name: CACHE, Help: "爬虫缓存", Value: kit.Data(
|
|
kit.MDB_SHORT, kit.MDB_LINK, kit.MDB_FIELD, "time,step,size,total,action,text,name,type,link",
|
|
kit.MDB_PATH, "usr/spide",
|
|
)},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
CACHE: {Name: "cache hash auto 添加 清理", Help: "爬虫缓存", Action: map[string]*ice.Action{
|
|
mdb.CREATE: {Name: "create", Help: "添加", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Option("_process", "_progress")
|
|
if m.Cmdy(mdb.SELECT, m.Prefix(CACHE), "", mdb.HASH, kit.MDB_LINK, m.Option(kit.MDB_LINK)); len(m.Appendv(kit.MDB_TOTAL)) > 0 {
|
|
return // 已经下载
|
|
}
|
|
m.Cmd(mdb.INSERT, m.Prefix(CACHE), "", mdb.HASH,
|
|
kit.MDB_LINK, m.Option(kit.MDB_LINK),
|
|
kit.MDB_TYPE, m.Option(kit.MDB_TYPE),
|
|
kit.MDB_NAME, m.Option(kit.MDB_NAME),
|
|
kit.MDB_TEXT, m.Option(kit.MDB_TEXT),
|
|
)
|
|
|
|
m.Option("progress", m.Prefix(CACHE), "", m.Option(kit.MDB_LINK))
|
|
msg := m.Cmd(web.SPIDE, web.SPIDE_DEV, web.SPIDE_CACHE, web.SPIDE_GET, m.Option(kit.MDB_LINK))
|
|
|
|
p := path.Join(m.Conf(m.Prefix(CACHE), kit.META_PATH), m.Option(kit.MDB_NAME))
|
|
m.Cmdy(nfs.LINK, p, msg.Append(kit.MDB_FILE))
|
|
|
|
if file, e := avutil.Open(p); e == nil {
|
|
defer file.Close()
|
|
|
|
if streams, e := file.Streams(); e == nil {
|
|
for _, stream := range streams {
|
|
|
|
switch {
|
|
case stream.Type().IsAudio():
|
|
case stream.Type().IsVideo():
|
|
// 自动分屏
|
|
vstream := stream.(av.VideoCodecData)
|
|
m.Cmd(nfs.LINK, path.Join(m.Conf(m.Prefix(CACHE), kit.META_PATH),
|
|
kit.Select("竖屏", "横屏", vstream.Width() > vstream.Height()), m.Option(kit.MDB_NAME)), p)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}},
|
|
mdb.REMOVE: {Name: "remove", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(mdb.DELETE, m.Prefix(CACHE), "", mdb.HASH, kit.MDB_HASH, m.Option(kit.MDB_HASH))
|
|
}},
|
|
mdb.PRUNES: {Name: "prunes", Help: "清理", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(mdb.PRUNES, m.Prefix(CACHE), "", mdb.HASH, kit.SSH_STEP, "100")
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.Option(ice.MSG_ACTION, "删除")
|
|
m.Option(mdb.FIELDS, m.Conf(m.Prefix(CACHE), kit.META_FIELD))
|
|
m.Cmdy(mdb.SELECT, m.Prefix(CACHE), "", mdb.HASH)
|
|
}},
|
|
},
|
|
}, nil)
|
|
}
|